def add(a, b):
    return a + b


def sub(a, b):
    return a - b


if __name__ == '__main__':
    b = {'ADD': add, 'SUB': sub}
    print(add(1, 2))
    print(b['ADD'](1, 2))
